import csv
import os

folder = '.'

# 创建输出文件夹
folder_path = "out"
if not os.path.exists(folder_path):
    os.makedirs(folder_path)

# 获取当前目录下的所有.csv文件
all_files = os.listdir(folder)
csv_files = []
for file in all_files:
    if file.endswith(".csv"):
        csv_files.append(file)

        # 读取 csv 文件
        with open(file, 'r', encoding='utf-8') as csvfile:
            csv_reader = csv.reader(csvfile)

            new_list = []  # 初始化新列表用于存储处理后的数据
            j = 1

            # 遍历 csv 文件中的每一行
            for i, row in enumerate(csv_reader):
                first_column = row[0]  # 获取第一列数据

                # 将第一列数据取出,变为5个新行
                new_row_1 = [first_column, j, 'ILF', 7]  # 名字对象
                j += 1
                new_row_2 = [first_column + '新增', j, 'EI', 4]
                j += 1
                new_row_3 = [first_column + '删除', j, 'EI', 4]
                j += 1
                new_row_4 = [first_column + '修改', j, 'EI', 4]
                j += 1
                new_row_5 = [first_column + '查询', j, 'EQ', 4]
                j += 1

                # 将新行添加到新列表中
                new_list.append(new_row_1)
                new_list.append(new_row_2)
                new_list.append(new_row_3)
                new_list.append(new_row_4)
                new_list.append(new_row_5)
        # 新文件名
        newName = 'new_' + file
        with open('./out/' + newName, 'w', encoding='gbk', newline='') as csvfile:
            csv_writer = csv.writer(csvfile)

            # 写入新的行
            for row in new_list:
                csv_writer.writerow(row)

# 打印 csv 文件列表
print(csv_files)
